NAME=mv_res_resnet18_hid512
LOG=./out/$NAME/log
SCORE=./out/$NAME/scores
OUT=./out/$NAME/scores
TSNE=./out/$NAME/tSNE
MODEL=./out/$NAME/weights
#####################################################
#mv_res_resnet18_hid512
#####################################################
CUDA_VISIBLE_DEVICES=0 \
python train.py --arch resnet18 --data-name ylimed \
    --data-root  $ROOT \
    --train-list $TrainSplit \
    --test-list  $PosValSplit \
    --representation mv \
    --model-prefix $MODEL/ylimed \
    --lr      0.005 \
    --lstm_lr 0.005 \
    --hidden_size 512 \
    --batch-size 80 \
    --lr-steps 80 120 \
    --epochs 160 \
    --num_segments 3 \
    --gpus 0 > $LOG/ylimed_mv_model.out 2>&1 &

CUDA_VISIBLE_DEVICES=3 \
python train.py --arch resnet18 --data-name ylimed \
    --data-root  $ROOT \
    --train-list $TrainSplit \
    --test-list  $PosValSplit \
    --representation residual \
    --model-prefix $MODEL/ylimed \
    --lr      0.001 \
    --lstm_lr 0.001 \
    --hidden_size 512 \
    --batch-size 80 \
    --lr-steps 80 110 \
    --epochs 130 \
    --num_segments 3 \
    --gpus 0 > $LOG/ylimed_residual_model.out 2>&1 &
#####################################################
CUDA_VISIBLE_DEVICES=0 \
python test.py \
    --gpus 0 \
    --arch resnet18 --data-name ylimed --representation residual \
    --hidden_size 512 \
    --data-root $ROOT \
    --test-list $TestSplit \
    --weights $MODEL/ylimed_residual_model_best.pth.tar \
    --save-scores $SCORE/ylimed_best_residual_model__scores \
    > $OUT/ylimed_best_residual_model__scores.out 2>&1 &
wait
